package com.example.xiaosen.repository;

import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @Author: zhouxj@ferrotec.com.cn
 * @version: v1.0
 * @Date: 2025/5/13 11:55
 * @Description:
 */
@Component
@RequiredArgsConstructor
public class InMemoryChatHistoryRepository implements ChatHistoryRepository{

    private final Map<String,List<String>> chatHistory = new HashMap<>();

    /**
     *保存会话记录，目前没有用户业务，因此没有根据用户划分，会话历史也只是简单存储在内存中，后续可以设计用户信息，并持久化到mysql、redis中
     *@param type 业务类型，如：chat、service、pdf
     *@param chatId 会话ID
     */
    @Override
    public void save(String type, String chatId) {
        //TODO: 添加用户字段进行分类存储
        List<String> chatIds = chatHistory.computeIfAbsent(type, k -> new ArrayList<>());
        if(!chatIds.contains(chatId)){
            chatIds.add(chatId);
        }
    }


    /**
     *获取会话ID列表
     *@param type 业务类型，如：chat、service、pdf
     *@return 会话ID列表
     */
    @Override
    public List<String> getChatIds(String type) {
        return chatHistory.getOrDefault(type,List.of());
    }
}
